package com.jianbing.util;

import javax.crypto.Cipher;
import javax.crypto.spec.IvParameterSpec;
import javax.crypto.spec.SecretKeySpec;

import java.security.spec.AlgorithmParameterSpec;

/**
 * Created by Han, Tixiang on 2016/11/17 0017.
 */
public class AESUtils {

    public  static String aesKey = "zt2i6cea0fceabb063e3961gg67op345";

    public static byte[] ivBytes = {0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00};

    // 加密
    public static String encrypt(String sSrc, String sKey) throws Exception {
        if (sSrc == null || sKey == null) {
            return null;
        }

        SecretKeySpec newKey = new SecretKeySpec(sKey.getBytes("UTF-8"), "AES");
        Cipher cipher = Cipher.getInstance("AES/CBC/PKCS5Padding");//"算法/模式/补码方式"
        // IvParameterSpec iv = new IvParameterSpec("0102030405060708".getBytes());//使用CBC模式，需要一个向量iv，可增加加密算法的强度
        AlgorithmParameterSpec ivSpec = new IvParameterSpec(ivBytes);

        cipher.init(Cipher.ENCRYPT_MODE, newKey, ivSpec);
        byte[] encrypted = cipher.doFinal(sSrc.getBytes());
        return Base64.encodeToString(encrypted, Base64.NO_PADDING | Base64.NO_WRAP);

        // return new BASE64Encoder().encode(encrypted);//此处使用BASE64做转码功能，同时能起到2次加密的作用。
    }



    // 解密
    public static String decrypt(String sSrc, String sKey) throws Exception {
        try {
            // 判断Key是否正确
            if (sSrc == null || sKey == null) {
                return null;
            }

            byte[] raw = sKey.getBytes("ASCII");
            SecretKeySpec skeySpec = new SecretKeySpec(raw, "AES");


            SecretKeySpec newKey = new SecretKeySpec(sKey.getBytes("UTF-8"), "AES");
            Cipher cipher = Cipher.getInstance("AES/CBC/PKCS5Padding");
          /*  IvParameterSpec iv = new IvParameterSpec("0102030405060708"
                    .getBytes());*/
            AlgorithmParameterSpec ivSpec = new IvParameterSpec(ivBytes);
            cipher.init(Cipher.DECRYPT_MODE, newKey, ivSpec);
            byte[] encrypted1 = Base64.decode(sSrc, Base64.NO_PADDING | Base64.NO_WRAP);
            //  byte[] encrypted1 = new BASE64Decoder().decodeBuffer(sSrc);//先用base64解密
            try {
                byte[] original = cipher.doFinal(encrypted1);
                String originalString = new String(original);
                return originalString;
            } catch (Exception e) {
                System.out.println(e.toString());
                return null;
            }
        } catch (Exception ex) {
            System.out.println(ex.toString());
            return null;
        }
    }

    public static void main(String[] args) throws Exception {
        /*
         * 加密用的Key 可以用26个字母和数字组成，最好不要用保留字符，虽然不会错，至于怎么裁决，个人看情况而定
         * 此处使用AES-128-CBC加密模式，key需要为16位。
         */
        String cKey = "zt2i6cea0fceabb063e3961gg67op345";
        // 需要加密的字串
        String cSrc = "{\"orderId\":\"100\",\"cuIdentity\":\"340824199010234227\",\"cuMobilePhone\":\"13221050002\",\"cuName\":\"abc\"}";
        System.out.println(cSrc);
        // 加密
        long lStart = System.currentTimeMillis();
        String enString = AESUtils.encrypt(cSrc, cKey);
        System.out.println(enString);
      //  System.out.println("加密后的字串是：" + enString.equals("9F6Hj7ydIwj2QAk/6vuMmA"));


        // 解密
        lStart = System.currentTimeMillis();
        String DeString = AESUtils.decrypt(enString, cKey);
        System.out.println("解密后的字串是：" + DeString);
    }
}
